package cn.cxyxj.config;

import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.List;

@Configuration
public class MybatisPlusConfig {

    /**
     * 需要忽略拼接多租户条件的表名
     */
    @Value("#{'${mybatis-plus.configuration.ignore-tenant-tables:}'.split(',')}")
    private List<String> ignoreTenantTables;

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {
            @Override
            public Expression getTenantId() {
                //获得当前登录用户的租户id
                return new LongValue(1111);
            }

            // 该 default 方法 默认返回 false 表示所有表都需要拼多租户条件
            // 如果有部分 sql 不需要加上租户ID条件
            // 可以使用 @InterceptorIgnore(tenantLine = "true") 标注在 Mapper 接口的方法上
            // 而 @SqlParser(filter = true) 在 mybatis-plus 3.4 版本中标记为过时
            @Override
            public boolean ignoreTable(String tableName) {
                return ignoreTenantTables.stream().anyMatch((e) -> e.equalsIgnoreCase(tableName));
            }
        }));
        return interceptor;
    }
}